package com.ymdd.library.pickerview.view;

import android.graphics.Typeface;
import android.view.View;
import com.ymdd.library.R;
import com.ymdd.library.pickerview.adapter.ArrayWheelAdapter;
import com.ymdd.library.pickerview.lib.WheelView;
import com.ymdd.library.pickerview.listener.OnItemSelectedListener;

import java.util.List;

public class WheelOptions<T> {
  private View view;
  private WheelView wv_option1;
  private WheelView wv_option2;
  private WheelView wv_option3;

  private List<T> mOptions1Items;
  private List<List<T>> mOptions2Items;
  private List<T> N_mOptions2Items;
  private List<List<List<T>>> mOptions3Items;
  private List<T> N_mOptions3Items;
  private boolean linkage;
  private OnItemSelectedListener wheelListener_option1;
  private OnItemSelectedListener wheelListener_option2;

  //文字的颜色和分割线的颜色
  int textColorOut;
  int textColorCenter;
  int dividerColor;

  private WheelView.DividerType dividerType;

  // 条目间距倍数
  float lineSpacingMultiplier = 1.6F;

  public View getView() {
    return view;
  }

  public void setView(View view) {
    this.view = view;
  }

  public WheelOptions(View view, Boolean linkage) {
    super();
    this.linkage = linkage;
    this.view = view;
    wv_option1 = (WheelView) view.findViewById(R.id.options1);// 初始化时显示的数据
    wv_option2 = (WheelView) view.findViewById(R.id.options2);
    wv_option3 = (WheelView) view.findViewById(R.id.options3);
  }

  public void setPicker(List<T> options1Items, List<List<T>> options2Items,
      List<List<List<T>>> options3Items) {
    this.mOptions1Items = options1Items;
    this.mOptions2Items = options2Items;
    this.mOptions3Items = options3Items;
    int len = ArrayWheelAdapter.DEFAULT_LENGTH;
    if (this.mOptions3Items == null) len = 8;
    if (this.mOptions2Items == null) len = 12;
    // 选项1
    wv_option1.setAdapter(new ArrayWheelAdapter(mOptions1Items, len));// 设置显示数据
    wv_option1.setCurrentItem(0);// 初始化时显示的数据
    // 选项2
    if (mOptions2Items != null) {
      wv_option2.setAdapter(new ArrayWheelAdapter(mOptions2Items.get(0)));// 设置显示数据
    }
    wv_option2.setCurrentItem(wv_option1.getCurrentItem());// 初始化时显示的数据
    // 选项3
    if (mOptions3Items != null) {
      wv_option3.setAdapter(new ArrayWheelAdapter(mOptions3Items.get(0).get(0)));// 设置显示数据
    }
    wv_option3.setCurrentItem(wv_option3.getCurrentItem());
    wv_option1.setIsOptions(true);
    wv_option2.setIsOptions(true);
    wv_option3.setIsOptions(true);

    if (this.mOptions2Items == null) wv_option2.setVisibility(View.GONE);
    if (this.mOptions3Items == null) wv_option3.setVisibility(View.GONE);

    // 联动监听器
    wheelListener_option1 = new OnItemSelectedListener() {

      @Override public void onItemSelected(int index) {
        int opt2Select = 0;
        if (mOptions2Items != null) {
          opt2Select = wv_option2.getCurrentItem();//上一个opt2的选中位置
          //新opt2的位置，判断如果旧位置没有超过数据范围，则沿用旧位置，否则选中最后一项
          opt2Select =
              opt2Select >= mOptions2Items.get(index).size() - 1 ? mOptions2Items.get(index).size()
                  - 1 : opt2Select;

          wv_option2.setAdapter(new ArrayWheelAdapter(mOptions2Items.get(index)));
          wv_option2.setCurrentItem(opt2Select);
        }
        if (mOptions3Items != null) {
          wheelListener_option2.onItemSelected(opt2Select);
        }
      }
    };
    wheelListener_option2 = new OnItemSelectedListener() {

      @Override public void onItemSelected(int index) {
        if (mOptions3Items != null) {
          int opt1Select = wv_option1.getCurrentItem();
          opt1Select =
              opt1Select >= mOptions3Items.size() - 1 ? mOptions3Items.size() - 1 : opt1Select;
          index =
              index >= mOptions2Items.get(opt1Select).size() - 1 ? mOptions2Items.get(opt1Select)
                  .size() - 1 : index;
          int opt3 = wv_option3.getCurrentItem();//上一个opt3的选中位置
          //新opt3的位置，判断如果旧位置没有超过数据范围，则沿用旧位置，否则选中最后一项
          opt3 = opt3 >= mOptions3Items.get(opt1Select).get(index).size() - 1 ? mOptions3Items.get(
              opt1Select).get(index).size() - 1 : opt3;

          wv_option3.setAdapter(
              new ArrayWheelAdapter(mOptions3Items.get(wv_option1.getCurrentItem()).get(index)));
          wv_option3.setCurrentItem(opt3);
        }
      }
    };

    // 添加联动监听
    if (options2Items != null && linkage) {
      wv_option1.setOnItemSelectedListener(wheelListener_option1);
    }
    if (options3Items != null && linkage) {
      wv_option2.setOnItemSelectedListener(wheelListener_option2);
    }
  }

  //不联动情况下
  public void setNPicker(List<T> options1Items, List<T> options2Items, List<T> options3Items) {
    this.mOptions1Items = options1Items;
    this.N_mOptions2Items = options2Items;
    this.N_mOptions3Items = options3Items;
    int len = ArrayWheelAdapter.DEFAULT_LENGTH;
    if (this.N_mOptions3Items == null) len = 8;
    if (this.N_mOptions2Items == null) len = 12;
    // 选项1
    wv_option1.setAdapter(new ArrayWheelAdapter(mOptions1Items, len));// 设置显示数据
    wv_option1.setCurrentItem(0);// 初始化时显示的数据
    // 选项2
    if (N_mOptions2Items != null) {
      wv_option2.setAdapter(new ArrayWheelAdapter(N_mOptions2Items));// 设置显示数据
    }
    wv_option2.setCurrentItem(wv_option1.getCurrentItem());// 初始化时显示的数据
    // 选项3
    if (N_mOptions3Items != null) {
      wv_option3.setAdapter(new ArrayWheelAdapter(N_mOptions3Items));// 设置显示数据
    }
    wv_option3.setCurrentItem(wv_option3.getCurrentItem());
    wv_option1.setIsOptions(true);
    wv_option2.setIsOptions(true);
    wv_option3.setIsOptions(true);

    if (this.N_mOptions2Items == null) wv_option2.setVisibility(View.GONE);
    if (this.N_mOptions3Items == null) wv_option3.setVisibility(View.GONE);
  }

  public void setTextContentSize(int textSize) {
    wv_option1.setTextSize(textSize);
    wv_option2.setTextSize(textSize);
    wv_option3.setTextSize(textSize);
  }

  private void setTextColorOut() {
    wv_option1.setTextColorOut(textColorOut);
    wv_option2.setTextColorOut(textColorOut);
    wv_option3.setTextColorOut(textColorOut);
  }

  private void setTextColorCenter() {
    wv_option1.setTextColorCenter(textColorCenter);
    wv_option2.setTextColorCenter(textColorCenter);
    wv_option3.setTextColorCenter(textColorCenter);
  }

  private void setDividerColor() {
    wv_option1.setDividerColor(dividerColor);
    wv_option2.setDividerColor(dividerColor);
    wv_option3.setDividerColor(dividerColor);
  }

  private void setDividerType() {
    wv_option1.setDividerType(dividerType);
    wv_option2.setDividerType(dividerType);
    wv_option3.setDividerType(dividerType);
  }

  private void setLineSpacingMultiplier() {
    wv_option1.setLineSpacingMultiplier(lineSpacingMultiplier);
    wv_option2.setLineSpacingMultiplier(lineSpacingMultiplier);
    wv_option3.setLineSpacingMultiplier(lineSpacingMultiplier);
  }

  /**
   * 设置选项的单位
   *
   * @param label1 单位
   * @param label2 单位
   * @param label3 单位
   */
  public void setLabels(String label1, String label2, String label3) {
    if (label1 != null) wv_option1.setLabel(label1);
    if (label2 != null) wv_option2.setLabel(label2);
    if (label3 != null) wv_option3.setLabel(label3);
  }

  /**
   * 设置是否循环滚动
   *
   * @param cyclic 是否循环
   */
  public void setCyclic(boolean cyclic) {
    wv_option1.setCyclic(cyclic);
    wv_option2.setCyclic(cyclic);
    wv_option3.setCyclic(cyclic);
  }

  /**
   * 设置字体样式
   *
   * @param font 系统提供的几种样式
   */
  public void setTypeface(Typeface font) {
    wv_option1.setTypeface(font);
    wv_option2.setTypeface(font);
    wv_option3.setTypeface(font);
  }

  /**
   * 分别设置第一二三级是否循环滚动
   *
   * @param cyclic1,cyclic2,cyclic3 是否循环
   */
  public void setCyclic(boolean cyclic1, boolean cyclic2, boolean cyclic3) {
    wv_option1.setCyclic(cyclic1);
    wv_option2.setCyclic(cyclic2);
    wv_option3.setCyclic(cyclic3);
  }

  /**
   * 返回当前选中的结果对应的位置数组 因为支持三级联动效果，分三个级别索引，0，1，2。
   * 在快速滑动未停止时，点击确定按钮，会进行判断，如果匹配数据越界，则设为0，防止index出错导致崩溃。
   *
   * @return 索引数组
   */
  public int[] getCurrentItems() {
    int[] currentItems = new int[3];
    currentItems[0] = wv_option1.getCurrentItem();

    if (mOptions2Items != null && mOptions2Items.size() > 0) {//非空判断
      currentItems[1] =
          wv_option2.getCurrentItem() > (mOptions2Items.get(currentItems[0]).size() - 1) ? 0
              : wv_option2.getCurrentItem();
    } else {
      currentItems[1] = wv_option2.getCurrentItem();
    }

    if (mOptions3Items != null && mOptions3Items.size() > 0) {//非空判断
      currentItems[2] = wv_option3.getCurrentItem() > (mOptions3Items.get(currentItems[0])
          .get(currentItems[1])
          .size() - 1) ? 0 : wv_option3.getCurrentItem();
    } else {
      currentItems[2] = wv_option3.getCurrentItem();
    }

    return currentItems;
  }

  public void setCurrentItems(int option1, int option2, int option3) {
    if (linkage) {
      itemSelected(option1, option2, option3);
    }
    wv_option1.setCurrentItem(option1);
    wv_option2.setCurrentItem(option2);
    wv_option3.setCurrentItem(option3);
  }

  private void itemSelected(int opt1Select, int opt2Select, int opt3Select) {
    if (mOptions2Items != null) {
      wv_option2.setAdapter(new ArrayWheelAdapter(mOptions2Items.get(opt1Select)));
      wv_option2.setCurrentItem(opt2Select);
    }
    if (mOptions3Items != null) {
      wv_option3.setAdapter(new ArrayWheelAdapter(mOptions3Items.get(opt1Select).get(opt2Select)));
      wv_option3.setCurrentItem(opt3Select);
    }
  }

  /**
   * 设置间距倍数,但是只能在1.2-2.0f之间
   */
  public void setLineSpacingMultiplier(float lineSpacingMultiplier) {
    this.lineSpacingMultiplier = lineSpacingMultiplier;
    setLineSpacingMultiplier();
  }

  /**
   * 设置分割线的颜色
   */
  public void setDividerColor(int dividerColor) {
    this.dividerColor = dividerColor;
    setDividerColor();
  }

  /**
   * 设置分割线的类型
   */
  public void setDividerType(WheelView.DividerType dividerType) {
    this.dividerType = dividerType;
    setDividerType();
  }

  /**
   * 设置分割线之间的文字的颜色
   */
  public void setTextColorCenter(int textColorCenter) {
    this.textColorCenter = textColorCenter;
    setTextColorCenter();
  }

  /**
   * 设置分割线以外文字的颜色
   */
  public void setTextColorOut(int textColorOut) {
    this.textColorOut = textColorOut;
    setTextColorOut();
  }

  /**
   * Label 是否只显示中间选中项的
   */

  public void isCenterLabel(Boolean isCenterLabel) {
    wv_option1.isCenterLabel(isCenterLabel);
    wv_option2.isCenterLabel(isCenterLabel);
    wv_option3.isCenterLabel(isCenterLabel);
  }
}
